package nconv.convert.plugins;

/**
 * Created by xiaobyu on 5/16/2016.
 */

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

public class P29215 extends DefaultTGPPConvertPlugin{
    Set<String> avpDefHeading = new HashSet<>(Arrays.asList("Subsession-Decision-Info", "Subsession-Enforcement-Info",
            "Subsession-Id", "Subsession-Operation", "Multiple-BBERF-Action", "DRA-Deployment", "DRA-Binding", "PCRF-Address"));

    @Override
    public String getReplacedAVPDefOrDiameterMessageHeading(String original) {
        if (avpDefHeading.contains(original.trim())) {
            return original + " AVP";
        }
        return original;
    }
}

